import pandas as pd

# 文件路径（请根据实际情况修改）
old_file = "market_data_20250804.csv"     # 原始数据（无简称）
code_name_file = "../basic_info/code_name.csv"  # 股票代码-简称表
output_file = "market_data_20250804_with_name.csv"  # 处理后输出文件

# 读取数据
df_data = pd.read_csv(old_file, dtype={"股票代码": str})
df_name = pd.read_csv(code_name_file, dtype={"股票代码": str})

# 合并股票简称（根据“股票代码”字段左连接）
df_merged = pd.merge(df_data, df_name, on="股票代码", how="left")

# 将“股票简称”列插入到“股票代码”后面
cols = df_merged.columns.tolist()
if "股票简称" in cols:
    # 移动“股票简称”列到“股票代码”后
    code_index = cols.index("股票代码")
    cols.insert(code_index + 1, cols.pop(cols.index("股票简称")))
    df_merged = df_merged[cols]

# 保存到新文件（你也可以覆盖原文件）
df_merged.to_csv(output_file, index=False, encoding="utf-8-sig")
print(f"已保存含简称数据到：{output_file}")
